var chat = function (server) {
    // 导入WebSocket模块:
    const WebSocket = require('ws');

    // 引用Server类:
    const WebSocketServer = WebSocket.Server;

    // 实例化:
    const wss = new WebSocketServer({
        server
    });
    const clients = [];

    wss.on("connection", function (ws) {
        clients.push(ws);
        ws.wss = wss;
        ws.on("message", function (msg) {
            msg = msg.toString();
            msg = JSON.parse(msg);
            console.log(5555, msg);
            if (msg.type == "login") {
                sendAll(msg.user + "加入聊天室");
            } else if (msg.msg) {
                sendAll(msg.msg)
            }
        });
    });
    wss.on("close", function (ws) {
        console.log("有人走了");
    });

    function sendAll(msg) {
        for (var i = 0; i < clients.length; i++) {
            clients[i].send(msg);
        }
    }

}
module.exports = chat